import Cocoa import SwiftUI class MapTextEditorController: NSViewController { @Binding var document: MapDocument let onChange: () -> Void private let vertexRegex = MapParsingPatterns.vertex private let edgeRegex = MapParsingPatterns.edge private let blockerRegex = MapParsingPatterns.blocker private let opportunityRegex = MapParsingPatterns.opportunity private let noteRegex = MapParsingPatterns.note private let stageRegex = MapParsingPatterns.stage private let groupRegex = MapParsingPatterns.group private let changeDebouncer: Debouncer = Debouncer(seconds: 1) init(document: Binding, onChange: @escaping () -> Void) { self._document = document self.onChange = onChange super.init(nibName: nil, bundle: nil) } required init?(coder: NSCoder) { fatalError("init(coder:) has not been implemented") } override func loadView() { let scrollView = NSTextView.scrollableTextView() let textView = scrollView.documentView as! NSTextView scrollView.translatesAutoresizingMaskIntoConstraints = false textView.backgroundColor = .ui.background textView.allowsUndo = true textView.delegate = self textView.textStorage?.delegate = self textView.string = self.document.text textView.isEditable = true textView.font = .monospacedSystemFont(ofSize: 16.0, weight: .regular) self.view = scrollView } override func viewDidAppear() { self.view.window?.makeFirstResponder(self.view) } } extension MapTextEditorController: NSTextViewDelegate { func textDidChange(_ obj: Notification) { if let textField = obj.object as? NSTextView { self.document.text = textField.string changeDebouncer.debounce { DispatchQueue.main.async { self.onChange() } } } } func textView(_ view: NSTextView, shouldChangeTextIn: NSRange, replacementString: String?) -> Bool { let range = Range(shouldChangeTextIn, in: view.string) let target = view.string[range!] if target == "--" { return false } return true } } extension MapTextEditorController: NSTextStorageDelegate { override func textStorageDidProcessEditing(_ obj: Notification) { if let textStorage = obj.object as? NSTextStorage { self.colorizeText(textStorage: textStorage) } } private func colorizeText(textStorage: NSTextStorage) { let range = NSMakeRange(0, textStorage.length) var matches = vertexRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.vertex], range: match.range(at: 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.number], range: match.range(at: 2)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.number], range: match.range(at: 3)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.option], range: match.range(at: 4)) } matches = edgeRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.vertex], range: match.range(at: 1)) let arrowRange = match.range(at: 2) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.symbol], range: NSMakeRange(arrowRange.lowerBound - 1, arrowRange.length + 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.vertex], range: match.range(at: 3)) } matches = opportunityRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.option], range: match.range(at: 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.vertex], range: match.range(at: 2)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.symbol], range: match.range(at: 3)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.number], range: match.range(at: 4)) } matches = blockerRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.option], range: match.range(at: 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.vertex], range: match.range(at: 2)) } matches = noteRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.option], range: match.range(at: 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.number], range: match.range(at: 2)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.number], range: match.range(at: 3)) } matches = stageRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.option], range: match.range(at: 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.number], range: match.range(at: 2)) } matches = groupRegex.matches(in: textStorage.string, options: [], range: range) for match in matches { textStorage.addAttributes( [.foregroundColor: NSColor.syntax.option], range: match.range(at: 1)) textStorage.addAttributes( [.foregroundColor: NSColor.syntax.vertex], range: match.range(at: 2)) } } } struct MapTextEditor: NSViewControllerRepresentable { @Binding var document: MapDocument var onChange: () -> Void = {} func makeNSViewController( context: NSViewControllerRepresentableContext ) -> MapTextEditorController { return MapTextEditorController(document: $document, onChange: onChange) } func updateNSViewController( _ nsViewController: MapTextEditorController, context: NSViewControllerRepresentableContext ) {} }